"""
所有模型的运行入口程序
"""
import os
from lr import lr_run
from logistic_reg import logistic_run
from text_sequences.lstm_model_pm25 import lstm_pm25_run
from classific.multi_class_full_nn import multi_class_full_nn_run
from classific.multi_class_cnn import multi_class_cnn_run
from vision.conv_lstm import conv_lstm_run
from classific.sequence_lstm import sequence_lstm_run
from classific.sequence_gru import sequence_gru_run
from vision.conv_cifar import conv_cifar_run


if __name__ == '__main__':
    gpu = False
    if gpu:
        os.environ['CUDA_DEVICE_ORDER'] = 'PCI_BUS_ID'
        os.environ['CUDA_VISIBLE_DEVICES'] = '0'
    else:
        os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
    workers = 8
    
    # 线性回归
    # lr_run(workers=workers)

    # 逻辑回归
    # logistic_run(workers=workers)

    # pm2.5的lstm
    # lstm_pm25_run(workers=workers)

    # mnist手写数字
    # multi_class_full_nn_run(workers=workers)
    # multi_class_cnn_run(workers=workers)

    # 卷积LSTM
    conv_lstm_run(workers=workers)

    # sequence lstm
    # sequence_lstm_run(workers=workers)

    # sequence gru
    # sequence_gru_run(workers=workers)

    # cifar10 卷积
    # conv_cifar_run(workers=workers)
